Simple Collision Detection And Input Handling In Pygame